import os
from custom_layer import MyDense
from keras import Sequential

if __name__ == '__main__':
    os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

    model = Sequential(
        [
            MyDense(28* 28, 256),
            MyDense(256, 128),
            MyDense(128,32),
            MyDense(32, 10),
        ]
    )

    model.build(input_shape=[None, 28* 28])
    model.summary()

    # 模型装配和训练
    # 模型预测